import dialogue from '../../utils/ai'
import { getUserInfo } from '../../api/index.js'

Page({
  data: {
    ttsWSRef: '',
    reply: '',
    isEnd: true,
    inputText: '',
    userInfo: {},
    sessionProcedure: []
  },

  // 大模型问答数据存储
  onMessageReceived: function () {
    // 合并数据更新
    this.setData({
      reply: '',
      sessionProcedure: this.data.sessionProcedure.concat({
        question: this.data.inputText,
        answer: '',
        finish: false
      }),
      inputText: ''
    })
  },

  dataRefresh: function (data) {
    this.setData({ reply: data })
    // 获取 sessionProcedure 数组的最后一个元素
    const lastIndex = this.data.sessionProcedure.length - 1
    if (lastIndex >= 0) {
      // 更新最后一个元素的 answer 属性
      const newAnswer = this.data.reply
      const newSessionProcedure = [...this.data.sessionProcedure]
      newSessionProcedure[lastIndex].answer = newAnswer

      // 使用 setData 更新数据
      this.setData({
        sessionProcedure: newSessionProcedure
      })
    }
  },

  // 判断大模型是否回答完毕
  endofJudgment(data) {
    const sessionProcedure = this.data.sessionProcedure
    if (sessionProcedure.length > 0) {
      // 直接在原始数据上修改
      sessionProcedure[sessionProcedure.length - 1].finish = true
      // 使用 setData 更新页面数据
      this.setData({
        sessionProcedure: sessionProcedure
      })
      this.setData({ isEnd: data })
    }
  },

  handleInput(event) {
    this.setData({ inputText: event.detail })
  },

  // 提问大模型
  handleAskModel() {
    dialogue(this.data.inputText, this.data.ttsWSRef, this.data.reply, this.onMessageReceived.bind(this), this.dataRefresh.bind(this), this.endofJudgment.bind(this))
  },

  // 获取用户的个人信息
  async userInfo() {
    const res = await getUserInfo()

    if (res.code === 200) {
      this.setData({ userInfo: res.data })
    }
  },

  /**
   * 生命周期函数--监听页面加载
   */
  onLoad(options) {
    this.userInfo()
  },

  /**
   * 生命周期函数--监听页面初次渲染完成
   */
  onReady() {},

  /**
   * 生命周期函数--监听页面显示
   */
  onShow() {},

  /**
   * 生命周期函数--监听页面隐藏
   */
  onHide() {},

  /**
   * 生命周期函数--监听页面卸载
   */
  onUnload() {},

  /**
   * 页面相关事件处理函数--监听用户下拉动作
   */
  onPullDownRefresh() {
    setTimeout(() => {
      wx.stopPullDownRefresh()
      wx.reLaunch({
        url: '/pages/index/index'
      })
    }, 300)
  },

  /**
   * 页面上拉触底事件的处理函数
   */
  onReachBottom() {},

  /**
   * 用户点击右上角分享
   */
  onShareAppMessage() {}
})
